package littlecyberwar.model;

import java.awt.*;
import javax.swing.*;

import littlecyberwar.ui.*;
/**
 * This is a container class that holds the unit name and the unit ImageIcon
 * In the future this will hold the view for a whole UNIT (multi-stand or
 * single stand)
 */
public class UnitTrayView {
  String name;
  Element unitElement;

  public UnitTrayView(String n, Element el) {
    name = n;
    unitElement = el;
  }

  public ImageIcon getUnitIcon() {
	  if ( unitElement != null ){
		  return ( unitElement.getImageIcon() );
	  } else {
		  return ( null );
	  }
  }

  public String getUnitName() {
    return ( name );
  }

  public Element getElement() {
    return ( unitElement );
  }

  /*
  public Unit getUnit() {
    return();
  }
  */

}